fixed a bug
Fixed bug