diff --git a/var/spack/repos/builtin/packages/ollama/package.py b/var/spack/repos/builtin/packages/ollama/package.py
index acd35240671..0c916b90bde 100644
--- a/var/spack/repos/builtin/packages/ollama/package.py
+++ b/var/spack/repos/builtin/packages/ollama/package.py
@@ -44,7 +44,7 @@ def setup_build_environment(self, env):
             cuda_prefix = self.spec["cuda"].prefix
             env.set("CUDACXX", cuda_prefix.bin.nvcc)
             env.set("CUDA_LIB_DIR", cuda_prefix.lib)
-            env.set("CMAKE_CUDA_ARCHITECTURES", spec.variants["cuda_arch"].value)
+            env.set("CMAKE_CUDA_ARCHITECTURES", self.spec.variants["cuda_arch"].value)
 
     @property
     def generate_args(self):