Graph tests - fix to prevent test failures