Some minor fixes to gravity prototyping.