Small bug fix
Fixed bug