resolving small bug