diff --git a/pisi/specfile.py b/pisi/specfile.py
index fac59e8e..ccd1e38b 100644
--- a/pisi/specfile.py
+++ b/pisi/specfile.py
@@ -95,5 +95,5 @@ class SpecFile(XmlFile):
     def write(self, filename):
         """Write PSPEC file"""
         self.newDOM()
-        self.addNodeText("Source/Name", source.name)
+        self.addNodeText("Source/Name", self.source.name)
         self.writexml(filename)
diff --git a/tests/specfiletests.py b/tests/specfiletests.py
index 9783e863..be7e07b8 100644
--- a/tests/specfiletests.py
+++ b/tests/specfiletests.py
@@ -3,6 +3,7 @@ import unittest
 import os
 
 from pisi import specfile
+from pisi.context import ctx
 
 class SpecFileTestCase(unittest.TestCase):
     def setUp(self):