Small bug fix :D