From 3b02642b48de1288f80d6f252072793b1c88f621 Mon Sep 17 00:00:00 2001
From: Ken Lauer <152229072+ken-lauer@users.noreply.github.com>
Date: Fri, 20 Dec 2024 09:24:15 -0800
Subject: [PATCH] FIX: conversion coefficient in x/y dimensions

Co-authored-by: Christopher Mayes <31023527+ChristopherMayes@users.noreply.github.com>
---
 pmd_beamphysics/wavefront.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pmd_beamphysics/wavefront.py b/pmd_beamphysics/wavefront.py
index ffbf358..9f6e7a4 100644
--- a/pmd_beamphysics/wavefront.py
+++ b/pmd_beamphysics/wavefront.py
@@ -392,7 +392,7 @@ def conversion_coeffs(wavelength: float, dim: int) -> tuple[float, ...]:
     Theta-x, theta-y, omega.
     """
     k0 = 2.0 * np.pi / wavelength
-    return tuple([2.0 * np.pi / k0] * (dim - 1) + [2.0 * np.pi * HBAR_EV_M])
+    return tuple([wavelength] * (dim - 1) + [2.0 * np.pi * HBAR_EV_M])
 
 
 def cartesian_domain(