diff --git a/deepxde/data/pde_operator.py b/deepxde/data/pde_operator.py
index fb47dd803..75e08958b 100644
--- a/deepxde/data/pde_operator.py
+++ b/deepxde/data/pde_operator.py
@@ -278,7 +278,7 @@ def forward_call(trunk_input):
                 output = aux[0]((inputs[0], trunk_input))
                 if not is_multi_outputs:
                     return output.reshape(shape0 * shape1, 1)
-                elif is_multi_outputs:
+                else:
                     return output.reshape(shape0 * shape1, shape2)
 
             if not is_multi_outputs:
diff --git a/deepxde/gradients/jacobian.py b/deepxde/gradients/jacobian.py
index d9e00fd55..4cbe46087 100644
--- a/deepxde/gradients/jacobian.py
+++ b/deepxde/gradients/jacobian.py
@@ -2,7 +2,6 @@
 
 from .. import config
 from ..backend import backend_name
-from .. import backend as bkd
 
 class Jacobian(ABC):
     """Compute `Jacobian matrix <https://en.wikipedia.org/wiki/Jacobian_matrix_and_determinant>`_