diff --git a/src/generator/templates/api/lib/mbuild.mako b/src/generator/templates/api/lib/mbuild.mako
index eed433bdd1..5c622972c4 100644
--- a/src/generator/templates/api/lib/mbuild.mako
+++ b/src/generator/templates/api/lib/mbuild.mako
@@ -345,6 +345,7 @@ ${self._setter_fn(resource, method, m, p, part_prop, ThisType, c)}\
         request_value_type = request_value.id
 %>\
 <%block filter="test_block_filter">\
+use ${util.library_name()}::api::enums::*;
 ${capture(util.test_prelude) | hide_filter}\
 % if request_value:
 use ${util.library_name()}::api::${request_value_type};