Add missing if-statement