Fixed small bug