diff --git a/tasks/install-ubuntu.yml b/tasks/install-ubuntu.yml
index fb6fc74..bcd0a39 100644
--- a/tasks/install-ubuntu.yml
+++ b/tasks/install-ubuntu.yml
@@ -7,7 +7,7 @@
 - name: install driver packages
   apt:
     name: "{{ nvidia_driver_package_version | ternary(item+'='+nvidia_driver_package_version, item) }}"
-    state: "{{ nvidia_driver_package_version }}"
+    state: "{{ nvidia_driver_package_state }}"
     autoremove: "{{ nvidia_driver_package_state == 'absent' }}"
     purge: "{{ nvidia_driver_package_state == 'absent' }}"
   with_items: "{{ nvidia_driver_ubuntu_packages }}"