Fixed very small bug