diff --git a/test/TestLogisticRegression.cpp b/test/TestLogisticRegression.cpp
index aa728641f..83fe9aeb5 100644
--- a/test/TestLogisticRegression.cpp
+++ b/test/TestLogisticRegression.cpp
@@ -44,7 +44,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegression)
 #ifdef KOMPUTE_SHADER_FROM_STRING
         sq->record<kp::OpAlgoBase>(
           params, "test/shaders/glsl/test_logistic_regression.comp",
-          std::array<uint32_t, 3>(), std::vector<float>({5.0}));
+          kp::Workgroup(), kp::Constants({5.0}));
 #else
         sq->record<kp::OpAlgoBase>(
           params,
@@ -52,7 +52,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegression)
             kp::shader_data::shaders_glsl_logisticregression_comp_spv,
             kp::shader_data::shaders_glsl_logisticregression_comp_spv +
               kp::shader_data::shaders_glsl_logisticregression_comp_spv_len),
-          std::array<uint32_t, 3>(), std::vector<float>({5.0}));
+          kp::Workgroup(), kp::Constants({5.0}));
 #endif
 
         sq->record<kp::OpTensorSyncLocal>({ wOutI, wOutJ, bOut, lOut });
@@ -93,7 +93,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegressionManualCopy)
     uint32_t ITERATIONS = 100;
     float learningRate = 0.1;
 
-    std::vector<float> wInVec = { 0.001, 0.001 };
+    kp::Constants wInVec = { 0.001, 0.001 };
     std::vector<float> bInVec = { 0 };
 
     std::shared_ptr<kp::Tensor> xI{ new kp::Tensor({ 0, 1, 1, 1, 1 }) };
@@ -129,7 +129,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegressionManualCopy)
 #ifdef KOMPUTE_SHADER_FROM_STRING
         sq->record<kp::OpAlgoBase>(
           params, "test/shaders/glsl/test_logistic_regression.comp.spv",
-           std::array<uint32_t, 3>(), kp::Algorithm::SpecializationContainer{{(uint32_t)5}});
+           kp::Workgroup(), kp::Algorithm::SpecializationContainer{{(uint32_t)5}});
 #else
         sq->record<kp::OpAlgoBase>(
           params,
@@ -137,7 +137,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegressionManualCopy)
             kp::shader_data::shaders_glsl_logisticregression_comp_spv,
             kp::shader_data::shaders_glsl_logisticregression_comp_spv +
               kp::shader_data::shaders_glsl_logisticregression_comp_spv_len),
-           std::array<uint32_t, 3>(), std::vector<float>({5.0}));
+           kp::Workgroup(), kp::Constants({5.0}));
 #endif
 
         sq->record<kp::OpTensorSyncLocal>({ wOutI, wOutJ, bOut, lOut });
diff --git a/test/TestMultipleAlgoExecutions.cpp b/test/TestMultipleAlgoExecutions.cpp
index 127550d50..b5710f52c 100644
--- a/test/TestMultipleAlgoExecutions.cpp
+++ b/test/TestMultipleAlgoExecutions.cpp
@@ -379,13 +379,13 @@ TEST(TestMultipleAlgoExecutions, TestAlgorithmSpecialized)
 
             sq = mgr.sequence();
 
-            auto spec = std::vector<float>({5.0, 0.3});
+            auto spec = kp::Constants({5.0, 0.3});
 
             sq->begin();
             sq->record<kp::OpAlgoBase>(
               { tensorA, tensorB },
               std::vector<char>(shader.begin(), shader.end()),
-              std::array<uint32_t, 3>(), spec);
+              kp::Workgroup(), spec);
             sq->end();
 
             sq->eval();