Small bug fix