2 files changed, 2 insertions, 2 deletions
diff --git a/units/graphics/jobs.pxu b/units/graphics/jobs.pxu
index b56f7f4..3e82694 100644
--- a/units/graphics/jobs.pxu
+++ b/units/graphics/jobs.pxu
@@ -6,7 +6,7 @@ category_id: com.canonical.plainbox::graphics
 id: graphics/{index}_driver_version_{product_slug}
 command:
 source graphics_env {driver} {index}
- graphics_driver
+ graphics_driver.py
 estimated_duration: 0.5
 _description: Parses Xorg.0.log and discovers the running X driver and version for the {vendor} {product} graphics card
 _summary: Test X driver/version for {vendor} {product}
diff --git a/units/graphics/legacy.pxu b/units/graphics/legacy.pxu
index f76c4d3..2e002a9 100644
--- a/units/graphics/legacy.pxu
+++ b/units/graphics/legacy.pxu
@@ -1,7 +1,7 @@
 plugin: shell
 category_id: com.canonical.plainbox::graphics
 id: graphics/driver_version
-command: graphics_driver
+command: graphics_driver.py
 estimated_duration: 0.500
 _description: Parses Xorg.0.Log and discovers the running X driver and version
 _summary: Test X driver/version