Fixed a minor bug

Fixed a minor bug