Fixed small bug (?)