Fixing "rocking" in locomotives