diff --git a/waveorder/models/isotropic_fluorescent_thick_3d.py b/waveorder/models/isotropic_fluorescent_thick_3d.py
index 6c78f3d..88c171f 100644
--- a/waveorder/models/isotropic_fluorescent_thick_3d.py
+++ b/waveorder/models/isotropic_fluorescent_thick_3d.py
@@ -110,7 +110,10 @@ def _calculate_wrap_unsafe_transfer_function(
 
 def visualize_transfer_function(viewer, optical_transfer_function, zyx_scale):
     add_transfer_function_to_viewer(
-        viewer, torch.real(optical_transfer_function), zyx_scale, clim_factor=0.05
+        viewer,
+        torch.real(optical_transfer_function),
+        zyx_scale,
+        clim_factor=0.05,
     )
 
 
diff --git a/waveorder/models/isotropic_thin_3d.py b/waveorder/models/isotropic_thin_3d.py
index e96a499..5ba9d22 100644
--- a/waveorder/models/isotropic_thin_3d.py
+++ b/waveorder/models/isotropic_thin_3d.py
@@ -5,7 +5,6 @@
 from torch import Tensor
 
 from waveorder import optics, sampling, util
-from waveorder.visuals.napari_visuals import add_transfer_function_to_viewer
 
 
 def generate_test_phantom(