diff --git a/invokeai/frontend/web/src/features/nodes/util/graph/buildAdHocUpscaleGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graph/buildAdHocUpscaleGraph.ts
index f511da7fdb..25987c327a 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graph/buildAdHocUpscaleGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graph/buildAdHocUpscaleGraph.ts
@@ -1,15 +1,13 @@
 import type { RootState } from 'app/store/store';
 import { fetchModelConfigWithTypeGuard } from 'features/metadata/util/modelFetchingHelpers';
+import type { GraphType } from 'features/nodes/util/graph/generation/Graph';
+import { Graph } from 'features/nodes/util/graph/generation/Graph';
 import { getBoardField } from 'features/nodes/util/graph/graphBuilderUtils';
-import {
-  type ImageDTO,
-  type Invocation,
-  isSpandrelImageToImageModelConfig,
-  type NonNullableGraph,
-} from 'services/api/types';
+import type { ImageDTO } from 'services/api/types';
+import { isSpandrelImageToImageModelConfig } from 'services/api/types';
 import { assert } from 'tsafe';
 
-import { addCoreMetadataNode, getModelMetadataField, upsertMetadata } from './canvas/metadata';
+import { getModelMetadataField } from './canvas/metadata';
 import { SPANDREL } from './constants';
 
 type Arg = {
@@ -17,32 +15,26 @@ type Arg = {
   state: RootState;
 };
 
-export const buildAdHocUpscaleGraph = async ({ image, state }: Arg): Promise<NonNullableGraph> => {
+export const buildAdHocUpscaleGraph = async ({ image, state }: Arg): Promise<GraphType> => {
   const { simpleUpscaleModel } = state.upscale;
 
   assert(simpleUpscaleModel, 'No upscale model found in state');
 
-  const upscaleNode: Invocation<'spandrel_image_to_image'> = {
+  const g = new Graph('adhoc-upscale-graph');
+  g.addNode({
     id: SPANDREL,
     type: 'spandrel_image_to_image',
     image_to_image_model: simpleUpscaleModel,
     image,
     board: getBoardField(state),
-  };
+    is_intermediate: false,
+  });
 
-  const graph: NonNullableGraph = {
-    id: `adhoc-upscale-graph`,
-    nodes: {
-      [SPANDREL]: upscaleNode,
-    },
-    edges: [],
-  };
   const modelConfig = await fetchModelConfigWithTypeGuard(simpleUpscaleModel.key, isSpandrelImageToImageModelConfig);
 
-  addCoreMetadataNode(graph, {}, SPANDREL);
-  upsertMetadata(graph, {
+  g.upsertMetadata({
     upscale_model: getModelMetadataField(modelConfig),
   });
 
-  return graph;
+  return g.getGraph();
 };