1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnn/test/ProfilerTests.cpp b/src/armnn/test/ProfilerTests.cpp
index 9376fa4cea..40e9f470ad 100644
--- a/src/armnn/test/ProfilerTests.cpp
+++ b/src/armnn/test/ProfilerTests.cpp
@@ -57,6 +57,8 @@ void RegisterUnregisterProfilerSingleThreadImpl(bool &res)
 
     // Check that the profiler has been un-registered for this thread.
     res &= !profilerManager.GetProfiler();
+
+    armnn::ProfilerManager::GetInstance().RegisterProfiler(nullptr);
 }
 
 } // namespace
@@ -351,6 +353,7 @@ BOOST_AUTO_TEST_CASE(ProfilerJsonPrinter)
                               "unit\": \"us\"\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n");
 
     BOOST_CHECK(output == blessedOutput);
+    armnn::ProfilerManager::GetInstance().RegisterProfiler(nullptr);
 }
 
 BOOST_AUTO_TEST_SUITE_END();