Fixed a minor bug