diff --git a/vintage/citrus-java-dsl/src/main/java/com/consol/citrus/dsl/builder/HttpActionBuilder.java b/vintage/citrus-java-dsl/src/main/java/com/consol/citrus/dsl/builder/HttpActionBuilder.java
index b00e9f3899..f57d029fd1 100644
--- a/vintage/citrus-java-dsl/src/main/java/com/consol/citrus/dsl/builder/HttpActionBuilder.java
+++ b/vintage/citrus-java-dsl/src/main/java/com/consol/citrus/dsl/builder/HttpActionBuilder.java
@@ -2,6 +2,7 @@
 
 import com.consol.citrus.TestAction;
 import com.consol.citrus.TestActionBuilder;
+import com.consol.citrus.endpoint.Endpoint;
 import com.consol.citrus.http.client.HttpClient;
 import com.consol.citrus.http.server.HttpServer;
 import com.consol.citrus.spi.ReferenceResolver;
@@ -26,7 +27,7 @@ public HttpClientActionBuilder client(String httpClient) {
         return new HttpClientActionBuilder(delegate.client(httpClient));
     }
 
-    public HttpServerActionBuilder server(HttpServer httpServer) {
+    public HttpServerActionBuilder server(Endpoint httpServer) {
         return new HttpServerActionBuilder(delegate.server(httpServer));
     }