Fix if_statement test