diff --git a/recipe.json b/recipe.json
index c3538e7..ec73a83 100644
--- a/recipe.json
+++ b/recipe.json
@@ -111,7 +111,10 @@
                 {
                     "if": "nvidia",
                     "type": "command",
-                    "commands": ["_ubuntu-drivers install --recommended"]
+                    "commands": [
+                        "apt install -y linux-headers-$(uname -r)",
+                        "ubuntu-drivers install --recommended"
+                    ]
                 }
             ]
         },