diff --git a/src/java/com/threerings/presents/tools/InvocationTask.java b/src/java/com/threerings/presents/tools/InvocationTask.java index 679f28b6d..4f408e65a 100644 --- a/src/java/com/threerings/presents/tools/InvocationTask.java +++ b/src/java/com/threerings/presents/tools/InvocationTask.java @@ -52,6 +52,7 @@ import com.samskivert.util.StringUtil; import com.samskivert.velocity.VelocityUtil; import com.threerings.presents.annotation.TransportHint; +import com.threerings.presents.net.Transport; import com.threerings.presents.client.InvocationService.InvocationListener; /** @@ -129,6 +130,11 @@ public abstract class InvocationTask extends Task imports.add(args[ii]); } + + // import Transport if used + if (!StringUtil.isBlank(getTransport())) { + imports.add(Transport.class); + } } public String getCode ()