diff --git a/ODLA/platforms/tensorrt/plugins/CMakeLists.txt b/ODLA/platforms/tensorrt/plugins/CMakeLists.txt
index e33315918..8b25cca6c 100644
--- a/ODLA/platforms/tensorrt/plugins/CMakeLists.txt
+++ b/ODLA/platforms/tensorrt/plugins/CMakeLists.txt
@@ -26,6 +26,7 @@ else()
       61
       70
       75
+      80
     )
   message(STATUS "GPU_ARCHS is not defined. Generating CUDA code for default SMs: ${GPU_ARCHS}")
 endif()
diff --git a/tests/unittests/lit_cases/test_tensorrt/test_and2d_tensorrt.cc b/tests/unittests/lit_cases/test_tensorrt/test_and2d_tensorrt.cc
index 395d9dd93..739338db7 100644
--- a/tests/unittests/lit_cases/test_tensorrt/test_and2d_tensorrt.cc
+++ b/tests/unittests/lit_cases/test_tensorrt/test_and2d_tensorrt.cc
@@ -23,7 +23,7 @@
 // RUN: %halo_compiler -target cxx -batch-size 1 %halo_compile_flags %data_path/test_and2d/model.onnx -o %t.cc
 // RUN: %cxx -c -fPIC -o %t.o %t.cc -I%odla_path/include
 // RUN: %cxx -g %s %t.o %t.bin -I%T -I%odla_path/include -I%unittests_path -I%data_path/test_and2d/test_data_set_0 %odla_link %device_link -lodla_tensorrt -o %t_tensorrt.exe -Wno-deprecated-declarations
-// RUN: ODLA_TRT_USE_EXPLICIT_BATCH=1 %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_and2d | FileCheck %s
+// RUN: %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_and2d | FileCheck %s
 // CHECK: Result Pass
 // clang-format on
 
diff --git a/tests/unittests/lit_cases/test_tensorrt/test_and3d_tensorrt.cc b/tests/unittests/lit_cases/test_tensorrt/test_and3d_tensorrt.cc
index c4e0573c6..0ed790722 100644
--- a/tests/unittests/lit_cases/test_tensorrt/test_and3d_tensorrt.cc
+++ b/tests/unittests/lit_cases/test_tensorrt/test_and3d_tensorrt.cc
@@ -23,7 +23,7 @@
 // RUN: %halo_compiler -target cxx -batch-size 1 %halo_compile_flags %data_path/test_and3d/model.onnx -o %t.cc
 // RUN: %cxx -c -fPIC -o %t.o %t.cc -I%odla_path/include
 // RUN: %cxx -g %s %t.o %t.bin -I%T -I%odla_path/include -I%unittests_path -I%data_path/test_and3d/test_data_set_0 %odla_link %device_link -lodla_tensorrt -o %t_tensorrt.exe -Wno-deprecated-declarations
-// RUN: ODLA_TRT_USE_EXPLICIT_BATCH=1 %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_and3d | FileCheck %s
+// RUN: %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_and3d | FileCheck %s
 // CHECK: Result Pass
 // clang-format on
 
diff --git a/tests/unittests/lit_cases/test_tensorrt/test_not_3d_tensorrt.cc b/tests/unittests/lit_cases/test_tensorrt/test_not_3d_tensorrt.cc
index c5f78e718..87cdf3590 100644
--- a/tests/unittests/lit_cases/test_tensorrt/test_not_3d_tensorrt.cc
+++ b/tests/unittests/lit_cases/test_tensorrt/test_not_3d_tensorrt.cc
@@ -22,7 +22,7 @@
 // RUN: %halo_compiler -target cxx -batch-size 1 %halo_compile_flags %data_path/test_not_3d/model.onnx -o %t.cc
 // RUN: %cxx -c -fPIC -o %t.o %t.cc -I%odla_path/include
 // RUN: %cxx -g %s %t.o %t.bin -I%T -I%odla_path/include -I%unittests_path -I%data_path/test_not_3d/test_data_set_0 %odla_link %device_link -lodla_tensorrt -o %t_tensorrt.exe -Wno-deprecated-declarations
-// RUN: ODLA_TRT_USE_EXPLICIT_BATCH=1 %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_not_3d | FileCheck %s
+// RUN: %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_not_3d | FileCheck %s
 // CHECK: Result Pass
 // clang-format on