diff --git a/roles/ml-python-packages/files/requirements.txt b/roles/ml-python-packages/files/requirements.txt
index e0c328cad5afb1940548470a6d119ac8f7949fd2..290a36283f25e0d277ac804651f5d3a7adb7cf9a 100644
--- a/roles/ml-python-packages/files/requirements.txt
+++ b/roles/ml-python-packages/files/requirements.txt
@@ -3,3 +3,5 @@ matplotlib==3.4.*
 pandas==1.3.*
 Pillow==8.4.*
 scikit-learn==1.0.*
+seaborn==0.11.*
+plotly==5.4.*
diff --git a/roles/ml-python-packages/tasks/main.yml b/roles/ml-python-packages/tasks/main.yml
index 0c9078aa7ec792e19c6438d0087cba5b77429944..c28fe2a8fde7d2d165bc01b9cb20aaeaaced23bd 100644
--- a/roles/ml-python-packages/tasks/main.yml
+++ b/roles/ml-python-packages/tasks/main.yml
@@ -23,8 +23,8 @@
 
 
 
-- name: Check if tensorflow is installed
-  shell: ". /home/{{ target_user }}/.ml-venv/bin/activate && pip3 freeze | grep -q tensorflow==2.7"
+- name: Check if tensorflow is installed (from wheel file)
+  shell: ". /home/{{ target_user }}/.ml-venv/bin/activate && pip3 freeze | grep -q 'tensorflow @ file'"
   register: tfinstalled
   changed_when: "tfinstalled.rc != 0"
   failed_when: false