Improve neo4j test coverage