Small bug removed