diff --git a/src/api/traits.cc b/src/api/traits.cc
index 712b93065..5d05b14b6 100644
--- a/src/api/traits.cc
+++ b/src/api/traits.cc
@@ -15,7 +15,7 @@ Trait::Trait(TraitID id, TraitIdArray conflict_traits, TraitType type)
 
 FileMappingTrait::FileMappingTrait(
     std::string &filename, std::unordered_map<std::string, u64> &offset_map,
-    bool flush, TraitCallback flush_cb, bool load, TraitCallback load_cb)
+    TraitCallback flush_cb, TraitCallback load_cb)
     : Trait(FILE_TRAIT, TraitIdArray(), TraitType::FILE_MAPPING),
       flush_cb(flush_cb),
       load_cb(load_cb),
diff --git a/src/api/traits.h b/src/api/traits.h
index 99561f144..438b2bf26 100644
--- a/src/api/traits.h
+++ b/src/api/traits.h
@@ -37,8 +37,8 @@ struct FileMappingTrait : public Trait {
   std::string filename;
   std::unordered_map<std::string, u64> offset_map;
   FileMappingTrait(std::string &filename,
-                   std::unordered_map<std::string, u64> &offset_map, bool flush,
-                   TraitCallback flush_cb, bool load, TraitCallback load_cb);
+                   std::unordered_map<std::string, u64> &offset_map,
+                   TraitCallback flush_cb, TraitCallback load_cb);
   void onAttach(TraitInput &blob, Trait *trait);
   void onDetach(TraitInput &blob, Trait *trait);
   void onLink(TraitInput &blob, Trait *trait);
diff --git a/test/trait_test.cc b/test/trait_test.cc
index 0168e619f..98e9c815d 100644
--- a/test/trait_test.cc
+++ b/test/trait_test.cc
@@ -81,8 +81,8 @@ TEST_CASE("CustomTrait",
       file_vbucket.Link(blob_name, args.filename, ctx);
       offset_map.emplace(blob_name, std::stol(blob_name) * info.FILE_PAGE);
     }
-    auto trait = hermes::api::FileMappingTrait(fullpath_str, offset_map, false,
-                                               NULL, false, NULL);
+    auto trait =
+        hermes::api::FileMappingTrait(fullpath_str, offset_map, NULL, NULL);
     file_vbucket.Attach(&trait, ctx);
     file_vbucket.Delete(ctx);
     file_bucket.Destroy(ctx);