fix: Added missing 'end'