Fix stupid logic mistake