--- a/test/testir.py	Sun Dec 01 18:37:23 2013 +0100
+++ b/test/testir.py	Tue Dec 03 18:00:22 2013 +0100
@@ -1,10 +1,8 @@
 import unittest, os
 import sys
-import c3
 import ppci
 import ir
-import transform
-import optimize
+from ppci.transform import ConstantFolder
 
 
 class IrCodeTestCase(unittest.TestCase):
@@ -53,7 +51,7 @@
 class ConstantFolderTestCase(unittest.TestCase):
     def setUp(self):
         self.b = ir.Builder()
-        self.cf = transform.ConstantFolder()
+        self.cf = ConstantFolder()
         self.m = ir.Module('test')
         self.b.setModule(self.m)