Solved small bug