Add missing end keyword