diff --git a/odak/learn/wave/classical.py b/odak/learn/wave/classical.py
index 3f8e3009..ab2bf631 100644
--- a/odak/learn/wave/classical.py
+++ b/odak/learn/wave/classical.py
@@ -351,8 +351,8 @@ def get_impulse_response_fresnel_kernel(nu, nv, dx = 8e-6, wavelength = 515e-9,
     wxs = torch.linspace(- dx / 2., dx / 2., aperture_samples[0], device = device)
     wys = torch.linspace(- dx / 2., dx / 2., aperture_samples[1], device = device)
     h = torch.zeros(nu * scale, nv * scale, dtype = torch.complex64, device = device)
-    pxs = torch.linspace(- dx / 2., dx / 2., 5, device = device)
-    pys = torch.linspace(- dx / 2., dx / 2., 5, device = device)
+    pxs = torch.linspace(- dx / 2., dx / 2., aperture_samples[2], device = device)
+    pys = torch.linspace(- dx / 2., dx / 2., aperture_samples[3], device = device)
     for wx in tqdm(wxs):
         for wy in wys:
             for px in pxs: