Resolve small bug