adding missing if