diff --git a/test/kernels/bang/test_bang_codegen.cc b/test/kernels/bang/test_bang_codegen.cc
index 62c7f089..b0e8fef7 100644
--- a/test/kernels/bang/test_bang_codegen.cc
+++ b/test/kernels/bang/test_bang_codegen.cc
@@ -56,7 +56,7 @@ TEST(BANG_Codegen, run) {
     auto temp3 = sqrt->getOutput();
     auto mul = g->addOp<MulObj>(d, temp3, nullptr);
     auto temp4 = mul->getOutput();
-    auto softmax = g->addOp<SigmoidObj>(temp4, nullptr, 0);
+    auto softmax = g->addOp<SigmoidObj>(temp4, nullptr);
 
     // allocate BANG memory
     g->dataMalloc();
diff --git a/test/kernels/cuda/test_cuda_codegen.cc b/test/kernels/cuda/test_cuda_codegen.cc
index 385d06ad..92d438ff 100644
--- a/test/kernels/cuda/test_cuda_codegen.cc
+++ b/test/kernels/cuda/test_cuda_codegen.cc
@@ -57,7 +57,7 @@ TEST(CUDA_Codegen, run) {
     auto temp3 = sqrt->getOutput();
     auto mul = g->addOp<MulObj>(d, temp3, nullptr);
     auto temp4 = mul->getOutput();
-    auto softmax = g->addOp<SigmoidObj>(temp4, nullptr, 0);
+    auto softmax = g->addOp<SigmoidObj>(temp4, nullptr);
 
     // allocate CUDA memory
     g->dataMalloc();