Fix module name

Author: atalman

test/smoke_test/smoke_test.py

@@ -72,12 +72,13 @@ def smoke_test_cuda(package: str) -> None:
         for module in MODULES:
             imported_module = importlib.import_module(module["name"])
             # TBD for vision move extension module to private so it will
-            # be _extention. For audio add version return from the check
+            # be _extention.
+            version = "N/A"
             if module["extension"] == "extension":
                 version = imported_module.extension._check_cuda_version()
-                print(f"{module['name']} CUDA: {version}")
             else:
-                imported_module._extension._check_cuda_version()
+                version = imported_module._extension._check_cuda_version()
+            print(f"{module['name']} CUDA: {version}")
 
 
 def smoke_test_conv2d() -> None: