diff --git a/mediapipe/framework/BUILD b/mediapipe/framework/BUILD
index 0ccd05e77..721cacc95 100644
--- a/mediapipe/framework/BUILD
+++ b/mediapipe/framework/BUILD
@@ -272,6 +272,8 @@ cc_library(
     ],
     deps = [
         ":calculator_base",
+        ":calculator_context",
+        ":calculator_contract",
         ":calculator_graph",
         ":calculator_registry",
         ":counter_factory",
diff --git a/mediapipe/framework/calculator_framework.h b/mediapipe/framework/calculator_framework.h
index afb73fb30..8f193fde8 100644
--- a/mediapipe/framework/calculator_framework.h
+++ b/mediapipe/framework/calculator_framework.h
@@ -52,6 +52,8 @@
 #define MEDIAPIPE_FRAMEWORK_CALCULATOR_FRAMEWORK_H_
 
 #include "mediapipe/framework/calculator_base.h"
+#include "mediapipe/framework/calculator_context.h"
+#include "mediapipe/framework/calculator_contract.h"
 #include "mediapipe/framework/calculator_graph.h"
 #include "mediapipe/framework/calculator_registry.h"
 #include "mediapipe/framework/counter_factory.h"