Fix to the gravity test