diff --git a/test/smoke/flang-288613-2/Makefile b/test/smoke/flang-288613-2/Makefile
index 2c1d032424..25f5c5f444 100644
--- a/test/smoke/flang-288613-2/Makefile
+++ b/test/smoke/flang-288613-2/Makefile
@@ -7,7 +7,7 @@ TESTSRC_ALL  = $(TESTSRC_MAIN) $(TESTSRC_AUX)
 
 FLANG        ?= flang
 OMP_BIN      = $(AOMP)/bin/$(FLANG)
-CC           = $(OMP_BIN) $(VERBOSE)
+CC           = $(OMP_BIN) $(VERBOSE) -mmlir --enable-gpu-heap-alloc
 #-ccc-print-phases
 #"-\#\#\#"
 
diff --git a/test/smoke/flang-288613-2/flang-288613.f90 b/test/smoke/flang-288613-2/flang-288613.f90
index d52539e54a..8a7b9d17e4 100644
--- a/test/smoke/flang-288613-2/flang-288613.f90
+++ b/test/smoke/flang-288613-2/flang-288613.f90
@@ -16,7 +16,7 @@ end subroutine dec_val_dev
         integer i,j,k,l,m
         real(rstd), dimension(nsize,nsize,nsize) :: tmp
 
-!$omp target teams
+!$omp target teams distribute
         do i=1,nsize
 #define PRIVATE_ARRAY 1
 #ifdef PRIVATE_ARRAY
@@ -35,7 +35,7 @@ end subroutine dec_val_dev
                 end do
             end do
         end do
-!$omp end target teams
+!$omp end target teams distribute
     end subroutine _compute_dev
 
     subroutine compute_dev()
diff --git a/test/smoke/flang-288613-3/Makefile b/test/smoke/flang-288613-3/Makefile
index 2c1d032424..25f5c5f444 100644
--- a/test/smoke/flang-288613-3/Makefile
+++ b/test/smoke/flang-288613-3/Makefile
@@ -7,7 +7,7 @@ TESTSRC_ALL  = $(TESTSRC_MAIN) $(TESTSRC_AUX)
 
 FLANG        ?= flang
 OMP_BIN      = $(AOMP)/bin/$(FLANG)
-CC           = $(OMP_BIN) $(VERBOSE)
+CC           = $(OMP_BIN) $(VERBOSE) -mmlir --enable-gpu-heap-alloc
 #-ccc-print-phases
 #"-\#\#\#"
 
diff --git a/test/smoke/flang-288613/Makefile b/test/smoke/flang-288613/Makefile
index 2c1d032424..25f5c5f444 100644
--- a/test/smoke/flang-288613/Makefile
+++ b/test/smoke/flang-288613/Makefile
@@ -7,7 +7,7 @@ TESTSRC_ALL  = $(TESTSRC_MAIN) $(TESTSRC_AUX)
 
 FLANG        ?= flang
 OMP_BIN      = $(AOMP)/bin/$(FLANG)
-CC           = $(OMP_BIN) $(VERBOSE)
+CC           = $(OMP_BIN) $(VERBOSE) -mmlir --enable-gpu-heap-alloc
 #-ccc-print-phases
 #"-\#\#\#"
 
diff --git a/test/smoke/flang-288613/flang-288613.f90 b/test/smoke/flang-288613/flang-288613.f90
index bdd1f06705..dd020e7590 100644
--- a/test/smoke/flang-288613/flang-288613.f90
+++ b/test/smoke/flang-288613/flang-288613.f90
@@ -21,7 +21,7 @@ end subroutine dec_val_dev
         real(rstd) :: tmp
 #endif
 
-!$omp target teams
+!$omp target teams distribute
         do i=1,nsize
 !$omp parallel do simd private(tmp)
             do j=1,nsize
@@ -44,7 +44,7 @@ end subroutine dec_val_dev
                 end do
             end do
         end do
-!$omp end target teams
+!$omp end target teams distribute
     end subroutine _compute_dev
 
     subroutine compute_dev()