diff --git a/.github/workflows/R-CMD-check.yaml b/.github/workflows/R-CMD-check.yaml
index e50a4a9..607ae9a 100644
--- a/.github/workflows/R-CMD-check.yaml
+++ b/.github/workflows/R-CMD-check.yaml
@@ -66,7 +66,7 @@ jobs:
 
           conda_install(
             packages = c(
-            "tensorflow-cpu<=2.15",
+            "tensorflow<=2.15",
             "torch",
             "torcheval",
             "safetensors",
diff --git a/R/install_and_config.R b/R/install_and_config.R
index a3628e0..10047fe 100644
--- a/R/install_and_config.R
+++ b/R/install_and_config.R
@@ -73,7 +73,7 @@ install_py_modules<-function(envname="aifeducation",
         conda = "auto",
         pip = TRUE)
     } else {
-      if(utils::compareVersion(tf_version,"2.15")<=0){
+      if(utils::compareVersion(tf_version,"2.16")<0){
         reticulate::conda_install(
           packages = c(
             paste0("tensorflow<=",tf_version)),