diff --git a/opencl_fdfd/csr.py b/opencl_fdfd/csr.py
index e24e780..dd0f13b 100644
--- a/opencl_fdfd/csr.py
+++ b/opencl_fdfd/csr.py
@@ -54,9 +54,6 @@ def cg(a, b, max_iters=10000, err_thresh=1e-6, context=None, queue=None, verbose
     p_step = ops.create_p_step(context)
     dot = ops.create_dot(context)
 
-    def a_step(E, H, p, events):
-        return a_step_full(E, H, p, inv_dxes, oeps, invm, gpec, gpmc, Pl, Pr, events)
-
     '''
     Start the solve
     '''
diff --git a/opencl_fdfd/ops.py b/opencl_fdfd/ops.py
index 7d51e58..0a4ff87 100644
--- a/opencl_fdfd/ops.py
+++ b/opencl_fdfd/ops.py
@@ -253,6 +253,6 @@ def create_a_csr(context):
                                     arguments=', '.join((v_out_args, m_args, v_in_args)))
 
     def spmv(v_out, m, v_in, e):
-        return spmv_kernel(v_out, m.row_ptr, m.col_ind, m.data, v_in, wait_for=e)
+        return [spmv_kernel(v_out, m.row_ptr, m.col_ind, m.data, v_in, wait_for=e)]
 
     return spmv