From ef878f63f6c238ebbaba9d94fba509103f31c12a Mon Sep 17 00:00:00 2001
From: Tobba <tobias.haegermarck@gmail.com>
Date: Sun, 23 May 2021 00:07:20 +0200
Subject: [PATCH] extruder: Flush step generation before changing non-linear
 extrusion parameters

Signed-off-by: Tobias Haegermarck <tobias.haegermarck@gmail.com>
---
 klippy/kinematics/extruder.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/klippy/kinematics/extruder.py b/klippy/kinematics/extruder.py
index 673744907e4e..8dd39757a0fe 100644
--- a/klippy/kinematics/extruder.py
+++ b/klippy/kinematics/extruder.py
@@ -210,7 +210,11 @@ def cmd_SET_PRESSURE_ADVANCE(self, gcmd):
     def cmd_SET_NONLINEAR_EXTRUSION(self, gcmd):
         a = gcmd.get_float('A', self.nonlinear_a)
         b = gcmd.get_float('B', self.nonlinear_b)
+
+        toolhead = self.printer.lookup_object('toolhead')
+        toolhead.flush_step_generation()
         self._set_nonlinear_extrusion(a, b)
+
         msg = ("a: %.6f\n"
                "b: %.6f"
                % (a, b))