Fixed small bug in last update