Added gravity fix