diff --git a/src/setup-ghidra-source.cmake b/src/setup-ghidra-source.cmake
index c196d37d9..db17e4e1f 100644
--- a/src/setup-ghidra-source.cmake
+++ b/src/setup-ghidra-source.cmake
@@ -163,10 +163,11 @@ set(sleigh_deccore_source_list
   "${library_root}/paramid.cc"
   "${library_root}/unionresolve.cc"
 )
-#if("${sleigh_RELEASE_TYPE}" STREQUAL "HEAD")
-#  list(APPEND sleigh_deccore_source_list
-#  )
-#endif()
+if("${sleigh_RELEASE_TYPE}" STREQUAL "HEAD")
+  list(APPEND sleigh_deccore_source_list
+    "${library_root}/modelrules.cc"
+  )
+endif()
 
 set(sleigh_extra_source_list
   "${library_root}/callgraph.cc"
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index ba269997e..05f8e03ae 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -25,10 +25,11 @@ add_executable(sleigh_ghidra_test
   "${library_root}/../unittests/testmarshal.cc"
 )
 
-# if(sleigh_RELEASE_IS_HEAD)
-#   target_sources(sleigh_ghidra_test PRIVATE
-#   )
-# endif()
+if(sleigh_RELEASE_IS_HEAD)
+  target_sources(sleigh_ghidra_test PRIVATE
+    "${library_root}/../unittests/testparamstore.cc"
+  )
+endif()
 
 target_compile_features(sleigh_ghidra_test PRIVATE cxx_std_11)
 target_include_directories(sleigh_ghidra_test PRIVATE "${library_root}")