Small bug solved