Fixed problem with direct sum neighbours in gravity tree