Bug removed

Small bug fix