diff --git a/test/test_doc_features.py b/test/test_doc_features.py
index 9822605..70fa02e 100644
--- a/test/test_doc_features.py
+++ b/test/test_doc_features.py
@@ -142,10 +142,6 @@ class Abstract2 {
     return f(i1, i2);
 }
 
-template<class A, class B, class C = A>
-C multiply(A a, B b) {
-    return static_cast<C>(a * b);
-}
 
 //-----
 namespace Namespace {
@@ -714,6 +710,14 @@ def test09_templated_function(self):
 
         import cppjit
 
+        cppjit.cppdef("""
+
+template<class A, class B, class C = A>
+C multiply(A a, B b) {
+return static_cast<C>(a * b);
+}
+
+""")
         mul = cppjit.gbl.multiply
 
         assert "multiply" in cppjit.gbl.__dict__
diff --git a/test/test_lowlevel.py b/test/test_lowlevel.py
index 0208a5f..bcc7ac3 100644
--- a/test/test_lowlevel.py
+++ b/test/test_lowlevel.py
@@ -61,13 +61,14 @@ def test03_memory(self):
         """Memory allocation and free-ing"""
 
         import cppjit
+        from cppjit import ll
 
         # regular C malloc/free
         mem = cppjit.gbl.malloc(16)
         cppjit.gbl.free(mem)
 
         # typed styles
-        mem = cppjit.ll.malloc[int](self.N)
+        mem = ll.malloc[int](self.N)
         assert len(mem) == self.N
         assert not mem.__cpp_array__
         for i in range(self.N):