Adds some fixes to force_gravity_test