Fix conditional statement