Add missing if