First try : fixed little bug