diff --git a/dev/cuda/residual_forward.cu b/dev/cuda/residual_forward.cu
index 41b3e31..bbbcde2 100644
--- a/dev/cuda/residual_forward.cu
+++ b/dev/cuda/residual_forward.cu
@@ -102,7 +102,7 @@ void residual_forward(int kernel_num,
 // ----------------------------------------------------------------------------
 
 int main(int argc, char **argv) {
-    set_main();
+    setup_main();
 
     int B = 8;
     int T = 1024;