Fixes small bug