diff --git a/derive/src/endpoint.rs b/derive/src/endpoint.rs
index cbaec3b..0b5aaec 100644
--- a/derive/src/endpoint.rs
+++ b/derive/src/endpoint.rs
@@ -498,6 +498,7 @@ fn expand_endpoint_type(mut ty: EndpointType, attrs: AttributeArgs, fun: &ItemFn
 
 		Ok(quote! {
 			use ::gotham_restful::export::FutureExt as _;
+			use ::gotham_restful::gotham::state::FromState as _;
 			#state_block
 			async move {
 				#handle_content