diff --git a/miscplot/variability.py b/miscplot/variability.py
index dcd7244..1c7af73 100644
--- a/miscplot/variability.py
+++ b/miscplot/variability.py
@@ -348,7 +348,7 @@ def _mk_data(filename: str) -> None:
                 for qq in rng.standard_normal(size=100) * std + mean:
                     rows.append(dict(MeasurementType=mm, Device=dd, DeviceVariant=vv, MeasuredValue=qq))
     df = polars.DataFrame(rows)
-    df.write_csv()
+    df.write_csv(filename)
 
 
 if __name__ == '__main__':
diff --git a/miscplot/wmap.py b/miscplot/wmap.py
index 58ce050..4aba0f8 100644
--- a/miscplot/wmap.py
+++ b/miscplot/wmap.py
@@ -1,8 +1,11 @@
 from typing import Any, Callable
 
-from matplotlib.figure import Figure
-from matplotlib.axes import Axes
 from matplotlib import pyplot
+from matplotlib.axes import Axes
+from matplotlib.figure import Figure
+from matplotlib.patches import PathPatch
+from matplotlib.textpath import TextPath
+from matplotlib.transforms import Affine2D
 import numpy
 from numpy.typing import ArrayLike
 
@@ -49,7 +52,7 @@ def wafermap(
             label = xy2sn(xii, yii)
             tp = TextPath((xii, yii), label, size=0.2 * aspect_ratio)
             bbox = tp.get_extents()
-            xform = Affine2D.translate(-bbox.width / 2, -bbox.height / 2)
+            xform = Affine2D().translate(-bbox.width / 2, -bbox.height / 2)
             ax.add_patch(PathPatch(xform.transform_path(tp), color='black'))
 
 #    ax.add_patch(pyplot.Circle((0, 0), radius, color='gray', alpha=0.5))