diff --git a/test/call_tests.jl b/test/call_tests.jl
index 9ebd7bbc..954c9498 100644
--- a/test/call_tests.jl
+++ b/test/call_tests.jl
@@ -22,8 +22,8 @@
         ]
     end
     omodes = ContinuousNormalizingFlows.Mode[TrainMode(), TestMode()]
-    ndata_ = Int[4]
-    nvars_ = Int[2]
+    ndata_ = Int[2^6]
+    nvars_ = Int[2^3]
     aug_steers = Bool[false, true]
     inplaces = Bool[false, true]
     adb_list = AbstractDifferentiation.AbstractBackend[
diff --git a/test/fit_tests.jl b/test/fit_tests.jl
index a0a4d09a..9a0ccb7a 100644
--- a/test/fit_tests.jl
+++ b/test/fit_tests.jl
@@ -22,8 +22,8 @@
         ]
     end
     n_epochs_ = Int[2]
-    ndata_ = Int[4]
-    nvars_ = Int[2]
+    ndata_ = Int[2^6]
+    nvars_ = Int[2^3]
     aug_steers = Bool[false, true]
     inplaces = Bool[false, true]
     adtypes = ADTypes.AbstractADType[