Fixed a bug

Fixed a bug