Put into try instead and catch ValueError?