Fixed bug with units moving too far