Fixed non-gravity bug