update tests to include graph and use our custom registry