Small fix
Fixed a minor issue