diff --git a/cscs-checks/apps/tensorflow/tensorflow_check.py b/cscs-checks/apps/tensorflow/tensorflow_check.py
index ca2ffc0f28..3c4eb68874 100644
--- a/cscs-checks/apps/tensorflow/tensorflow_check.py
+++ b/cscs-checks/apps/tensorflow/tensorflow_check.py
@@ -14,7 +14,7 @@ def __init__(self, model_name):
         self.tags = {'production'}
         self.num_tasks = 1
         self.num_gpus_per_node = 1
-        self.modules = ['TensorFlow/1.7.0-CrayGNU-18.07-cuda-9.1-python3']
+        self.modules = ['TensorFlow/1.7.0-CrayGNU-18.08-cuda-9.1-python3']
 
         # Checkout to the branch corresponding to the module version of
         # TensorFlow