diff --git a/runner/trial/generator.d b/runner/trial/generator.d
index cc728f7..45873f1 100644
--- a/runner/trial/generator.d
+++ b/runner/trial/generator.d
@@ -85,6 +85,7 @@ string generateTestFile(Settings settings, bool hasTrialDependency, string[2][]
     import("settings.d") ~
     import("step.d") ~
     import("coverage.d") ~
+    import("attributes.d") ~
     import("stackresult.d");
 
   string code;