diff --git a/lib/testing/fake_agent_runner_storage.h b/lib/testing/fake_agent_runner_storage.h
index 79b46b7..224f110 100644
--- a/lib/testing/fake_agent_runner_storage.h
+++ b/lib/testing/fake_agent_runner_storage.h
@@ -8,7 +8,6 @@
 #include <functional>
 #include <string>
 
-#include <lib/fxl/functional/closure.h>
 #include <lib/fxl/macros.h>
 
 #include "peridot/bin/sessionmgr/agent_runner/agent_runner_storage.h"
@@ -22,7 +21,7 @@
 
   // |AgentRunnerStorage|
   void Initialize(NotificationDelegate* /*delegate*/,
-                  const fxl::Closure done) override {
+                  const std::function<void()> done) override {
     done();
   }