- fixed a small bug