Fix for missing 'end'

Oops