Fixed a small bug
Fixed a small bug