if/else fixes