diff --git a/isolator/test_isolator_module.cpp b/isolator/test_isolator_module.cpp
index d7f7d01..00c3b39 100644
--- a/isolator/test_isolator_module.cpp
+++ b/isolator/test_isolator_module.cpp
@@ -16,6 +16,8 @@
  * limitations under the License.
  */
 
+#include <string>
+
 #include <mesos/mesos.hpp>
 
 #include <mesos/module/isolator.hpp>
@@ -108,7 +110,8 @@ process::Future<ContainerLimitation> TestIsolatorProcess::watch(
 
 process::Future<Nothing> TestIsolatorProcess::update(
     const ContainerID& containerId,
-    const Resources& resources)
+    const Resources& resourceRequests,
+    const google::protobuf::Map<std::string, Value::Scalar>& resourceLimits = {})
 {
   if (!promises.contains(containerId)) {
     return process::Failure("Unknown container: " + stringify(containerId));
diff --git a/isolator/test_isolator_module.hpp b/isolator/test_isolator_module.hpp
index aa5dbfe..1d02da4 100644
--- a/isolator/test_isolator_module.hpp
+++ b/isolator/test_isolator_module.hpp
@@ -19,6 +19,8 @@
 #ifndef __TEST_ISOLATOR_MODULE_HPP__
 #define __TEST_ISOLATOR_MODULE_HPP__
 
+#include <string>
+
 #include <mesos/mesos.hpp>
 
 #include <mesos/slave/isolator.hpp>
@@ -61,7 +63,8 @@ class TestIsolatorProcess : public process::Process<TestIsolatorProcess>
 
   process::Future<Nothing> update(
       const ContainerID& containerId,
-      const Resources& resources);
+      const Resources& resourceRequests,
+      const google::protobuf::Map<std::string, Value::Scalar>& resourceLimits);
 
   process::Future<ResourceStatistics> usage(
       const ContainerID& containerId);
@@ -150,12 +153,15 @@ class TestIsolator : public mesos::slave::Isolator
 
   virtual process::Future<Nothing> update(
       const ContainerID& containerId,
-      const Resources& resources) override
+      const Resources& resourceRequests,
+      const google::protobuf::Map<
+          std::string, Value::Scalar>& resourceLimits = {}) override
   {
     return dispatch(process.get(),
                     &TestIsolatorProcess::update,
                     containerId,
-                    resources);
+                    resourceRequests,
+		    resourceLimits);
   }
 
   virtual process::Future<ResourceStatistics> usage(