small fix for gravity