[fix] Added missing end call