diff --git a/fdfd_tools/bloch.py b/fdfd_tools/bloch.py
index df63e6e..33e9255 100644
--- a/fdfd_tools/bloch.py
+++ b/fdfd_tools/bloch.py
@@ -68,7 +68,7 @@ This module contains functions for generating and solving the
 
     k, f = find_k(frequency=1/1550,
                   tolerance=(1/1550 - 1/1551),
-                  search_direction=[1, 0, 0],
+                  direction=[1, 0, 0],
                   G_matrix=recip_lattice,
                   epsilon=epsilon,
                   band=0)
@@ -369,7 +369,7 @@ def eigsolve(num_modes: int,
 
 def find_k(frequency: float,
            tolerance: float,
-           search_direction: numpy.ndarray,
+           direction: numpy.ndarray,
            G_matrix: numpy.ndarray,
            epsilon: field_t,
            mu: field_t = None,
@@ -380,7 +380,7 @@ def find_k(frequency: float,
 
     :param frequency: Target frequency.
     :param tolerance: Target frequency tolerance.
-    :param search_direction: k-vector direction to search along.
+    :param direction: k-vector direction to search along.
     :param G_matrix: 3x3 matrix, with reciprocal lattice vectors as columns.
     :param epsilon: Dielectric constant distribution for the simulation.
         All fields are sampled at cell centers (i.e., NOT Yee-gridded)
@@ -390,10 +390,10 @@ def find_k(frequency: float,
     return: (k, actual_frequency) The found k-vector and its frequency
     """
 
-    search_direction = numpy.array(search_direction) / norm(search_direction)
+    direction = numpy.array(direction) / norm(direction)
 
     def get_f(k0_mag: float, band: int = 0):
-        k0 = search_direction * k0_mag
+        k0 = direction * k0_mag
         n, _v = eigsolve(band + 1, k0, G_matrix=G_matrix, epsilon=epsilon)
         f = numpy.sqrt(numpy.abs(numpy.real(n[band])))
         return f
@@ -401,6 +401,6 @@ def find_k(frequency: float,
     res = scipy.optimize.minimize_scalar(lambda x: abs(get_f(x, band) - frequency), 0.25,
                                          method='Bounded', bounds=(0, 0.5),
                                          options={'xatol': abs(tolerance)})
-    return res.x * search_direction, res.fun + frequency
+    return res.x * direction, res.fun + frequency