Fixed Small bug