Fixed a small bug