Fix Neo4j tests failing