Fixed small bug, thanks to MakeALot