rust: Switch to a neater module based step function derive

Signed-off-by: Daniel Silverstone <dsilvers@digital-scurf.org>

1 files changed, 2 insertions, 2 deletions

diff --git a/templates/rust/template.rs.tera b/templates/rust/template.rs.tera
index a60599f..c972d37 100644
--- a/templates/rust/template.rs.tera
+++ b/templates/rust/template.rs.tera
@@ -29,7 +29,7 @@ lazy_static! {
 fn {{ scenario.title | nameslug }}() {
     let mut scenario = Scenario::new(&base64_decode("{{scenario.title | base64}}"));
     {% for step in scenario.steps %}
-    let step = BUILDER_{{step.function}}::default()
+    let step = {{step.function}}::Builder::default()
     {% for part in step.parts %}{% if part.CapturedText is defined -%}
         {%- set name = part.CapturedText.name -%}
         {%- set text = part.CapturedText.text -%}
@@ -58,7 +58,7 @@ fn {{ scenario.title | nameslug }}() {
     {% endfor -%}
         .build();
     {%- if step.cleanup %}
-    let cleanup = BUILDER_{{step.cleanup}}::default().build();
+    let cleanup = {{step.cleanup}}::Builder::default().build();
     scenario.add_step(step, Some(cleanup));
     {%- else %}
     scenario.add_step(step, None);