Small bug fix :)