Small bug fixed