Fixed a minor issue