Removed bug with if statement

Version 1.1.1