Fix for small bug