Add missing if statement