--- a/test/testasm.py	Sun Mar 02 17:12:08 2014 +0100
+++ b/test/testasm.py	Fri Mar 07 17:05:32 2014 +0100
@@ -1,8 +1,7 @@
 #!/usr/bin/python
 
-import unittest, cProfile
+import unittest
 from ppci import CompilerError
-from ppci.asmnodes import AInstruction, ABinop, AUnop, ASymbol, ALabel, ANumber
 from ppci.assembler import tokenize, Assembler, Lexer
 from ppci.objectfile import ObjectFile
 from ppci.linker import Linker
@@ -36,7 +35,7 @@
 
 
 class AssemblerParsingTestCase(unittest.TestCase):
-    """ 
+    """
         Tests the assembler parts
     """
     def setUp(self):
@@ -105,7 +104,7 @@
 class AsmTestCaseBase(unittest.TestCase):
     """ Base testcase for assembly """
     def feed(self, line):
-        self.a.assemble(line)
+        self.a.assemble(line, self.ostream)
 
     def check(self, hexstr):
         l = Linker()