Add missing end statement

Missed this in my conflict merge