How to prevent this type of error in python