fix gravity bug