if/else bugfix