1 files changed, 8 insertions, 4 deletions
diff --git a/src/backends/backendsCommon/test/MockBackend.cpp b/src/backends/backendsCommon/test/MockBackend.cpp
index 4bdb8ce786..a34cfe98fe 100644
--- a/src/backends/backendsCommon/test/MockBackend.cpp
+++ b/src/backends/backendsCommon/test/MockBackend.cpp
@@ -187,12 +187,16 @@ OptimizationViews MockBackend::OptimizeSubgraphView(const SubgraphView& subgraph
         {
             ARMNN_ASSERT(supportedSubgraph != nullptr);
 
-            PreCompiledLayer* preCompiledLayer =
-                optimizationViews.GetGraph().AddLayer<PreCompiledLayer>(
+            CompiledBlobPtr blobPtr;
+            BackendId backend = MockBackendId();
+
+            IConnectableLayer* preCompiledLayer =
+                optimizationViews.GetINetwork()->AddPrecompiledLayer(
                         PreCompiledDescriptor(supportedSubgraph->GetNumInputSlots(),
                                               supportedSubgraph->GetNumOutputSlots()),
-                        "pre-compiled");
-            preCompiledLayer->SetBackendId(MockBackendId());
+                                              blobPtr,
+                                              backend,
+                                              nullptr);
 
             SubgraphView substitutionSubgraph(*supportedSubgraph);
             SubgraphView replacementSubgraph(preCompiledLayer);