Use enum == instead of equals()