Small fix
Fixed bug