diff --git a/src/tensorrt/test/tensorrt_runtime.cpp b/src/tensorrt/test/tensorrt_runtime.cpp
index 69714042..a3362e09 100644
--- a/src/tensorrt/test/tensorrt_runtime.cpp
+++ b/src/tensorrt/test/tensorrt_runtime.cpp
@@ -243,7 +243,6 @@ TEST(TestOprTensorRT, IOFormatFree) {
     func2->execute();
     MGB_ASSERT_TENSOR_EQ(y1, y2);
 }
-#endif
 
 TEST(TestOprTensorRT, FlattenConcatPlugin) {
     REQUIRE_GPU(1);
@@ -272,6 +271,7 @@ TEST(TestOprTensorRT, FlattenConcatPlugin) {
     func->execute();
     MGB_ASSERT_TENSOR_EQ(host_z1, host_z2);
 }
+#endif
 
 TEST(TestOprTensorRT, ICudaEngine) {
     REQUIRE_GPU(1);