diff --git a/libraries/model-baker/src/model-baker/ParseFlowDataTask.cpp b/libraries/model-baker/src/model-baker/ParseFlowDataTask.cpp
index 6dff4f8c55..10991ecbe6 100644
--- a/libraries/model-baker/src/model-baker/ParseFlowDataTask.cpp
+++ b/libraries/model-baker/src/model-baker/ParseFlowDataTask.cpp
@@ -8,10 +8,11 @@
 
 #include "ParseFlowDataTask.h"
 
-void ParseFlowDataTask::run(const baker::BakeContextPointer& context, const Input& mapping, Output& output) {
+void ParseFlowDataTask::run(const baker::BakeContextPointer& context, const Input& mappingPair, Output& output) {
     FlowData flowData;
     static const QString FLOW_PHYSICS_FIELD = "flowPhysicsData";
     static const QString FLOW_COLLISIONS_FIELD = "flowCollisionsData";
+    auto mapping = mappingPair.second;
     for (auto mappingIter = mapping.begin(); mappingIter != mapping.end(); mappingIter++) {
         if (mappingIter.key() == FLOW_PHYSICS_FIELD || mappingIter.key() == FLOW_COLLISIONS_FIELD) {
             QByteArray data = mappingIter.value().toByteArray();
diff --git a/libraries/model-baker/src/model-baker/ParseFlowDataTask.h b/libraries/model-baker/src/model-baker/ParseFlowDataTask.h
index 7e1bc9fba1..65b8f7654b 100644
--- a/libraries/model-baker/src/model-baker/ParseFlowDataTask.h
+++ b/libraries/model-baker/src/model-baker/ParseFlowDataTask.h
@@ -12,9 +12,11 @@
 #include <hfm/HFM.h>
 #include "Engine.h"
 
+#include "BakerTypes.h"
+
 class ParseFlowDataTask {
 public:
-    using Input = QVariantHash;
+    using Input = baker::GeometryMappingPair;
     using Output = FlowData;
     using JobModel = baker::Job::ModelIO<ParseFlowDataTask, Input, Output>;