Fix gravity stuff