diff --git a/artifacts/nvidia.yml b/artifacts/nvidia.yml
index 90f91c2..922ec41 100644
--- a/artifacts/nvidia.yml
+++ b/artifacts/nvidia.yml
@@ -1,5 +1,7 @@
 ---
 - hosts: localhost
   connection: local
+  vars:
+    nvidia_driver_branch: "{{ nvidia_driver_branch }}"
   roles:
     - role: 'nvidia.nvidia_driver'
diff --git a/templates/nvidia.yml b/templates/nvidia.yml
index c84d701..4873eb9 100644
--- a/templates/nvidia.yml
+++ b/templates/nvidia.yml
@@ -33,7 +33,7 @@ topology_template:
       type: tosca.nodes.SoftwareComponent
       artifacts:
         nvidia_role:
-          file: NVIDIA.nvidia_driver
+          file: nvidia.nvidia_driver
           type: tosca.artifacts.AnsibleGalaxy.role
       requirements:
         - host: simple_node