diff --git a/defaults/main.yml b/defaults/main.yml
index b787dcc..41531a2 100644
--- a/defaults/main.yml
+++ b/defaults/main.yml
@@ -2,10 +2,15 @@
 # defaults file for hass
 
 hass_python_package:
-  - mysqlclient
   - homeassistant
   - myjdapi
+
+hass_python_postgres_package:
   - psycopg2
+
+hass_python_Mysql_package:
+  - mysqlclient
+
 hass_conf_folder: /var/lib/hass
 hass_venv_location: /opt/venv/hass
 hass_db_type: postgres #postgres or mysql
diff --git a/tasks/database_mysql.yml b/tasks/database_mysql.yml
index 21c1c5e..68b9f70 100644
--- a/tasks/database_mysql.yml
+++ b/tasks/database_mysql.yml
@@ -25,3 +25,11 @@
     password: "{{ hass_db_password }}"
     priv: "{{ hass_db_name }}.*:ALL"
     state: present
+
+- name: install Mysql python module
+  pip:
+    name: "{{hass_python_Mysql_package}}" 
+    virtualenv: "{{hass_venv_location}}"
+    virtualenv_command: /usr/bin/python -m venv
+  become: true
+  become_user: "{{hass_username}}"
diff --git a/tasks/database_postgres.yml b/tasks/database_postgres.yml
index 0e7d7d3..c984e22 100644
--- a/tasks/database_postgres.yml
+++ b/tasks/database_postgres.yml
@@ -6,6 +6,7 @@
     password: "{{ hass_db_password }}"
     state: present
 
+
 - name: "Ensure database is present"
   become: yes
   become_user: postgres
@@ -19,6 +20,7 @@
   register: hass_database_creation
 
 
+
 - name: import DATA in database in case of creation
   become: yes
   become_user: postgres
@@ -27,4 +29,12 @@
     state: restore
     target: "{{ hass_SQl_target_file }}"
   register: hass_database_import
-  when: hass_database_creation.changed == true and hass_SQl_target_file is defined
\ No newline at end of file
+  when: hass_database_creation.changed == true and hass_SQl_target_file is defined
+
+- name: install postgres python module
+  pip:
+    name: "{{hass_python_postgres_package}}" 
+    virtualenv: "{{hass_venv_location}}"
+    virtualenv_command: /usr/bin/python -m venv
+  become: true
+  become_user: "{{hass_username}}"