diff --git a/src/pystencils_autodiff/autodiff.py b/src/pystencils_autodiff/autodiff.py
index a864db501d8e484e3115771307dbd6be022ba4e5..984eaaf808102faebbbc96c6d264003a109ea214 100644
--- a/src/pystencils_autodiff/autodiff.py
+++ b/src/pystencils_autodiff/autodiff.py
@@ -21,11 +21,12 @@ class DiffModes(str, Enum):
 
 
 def get_jacobian_of_assignments(assignments, diff_variables):
-    """ Calculates the Jacobian of iterable of assignments wrt. to diff_variables
+    """
+    Calculates the Jacobian of iterable of assignments wrt. diff_variables
 
     Arguments:
-        assignments {[type]} -- [description]
-        diff_variables {[type]} -- [description]
+        assignments (List[pystencils.Assignment]): A collection of assignments or a AssignmentCollection
+        diff_variables (List[sympy.Symbol]): List of variables used to differentiate
 
     Returns:
         sp.Matrix -- Jacobian of statements
diff --git a/tests/lbm/adjoint_field.py b/tests/lbm/adjoint_field.py
index 9a598b5fc08e1577c4a8d25bffc05c8cd9bc30a6..3ac93927c91fd630417e671e96cbcd9ad05c794c 100644
--- a/tests/lbm/adjoint_field.py
+++ b/tests/lbm/adjoint_field.py
@@ -1,8 +1,10 @@
 import pystencils
 from pystencils.astnodes import FieldShapeSymbol, FieldStrideSymbol
 
-"""Determines how adjoint fields will be denoted in LaTeX output in terms of the forward field representation %s
-Default: r"\hat{%s}" """
+"""
+Determines how adjoint fields will be denoted in LaTeX output in terms of the forward field representation %s
+Default: "\\hat{%s}"
+"""
 ADJOINT_FIELD_LATEX_HIGHLIGHT = r"\hat{%s}"