Fixed a bug

fixed a bug