I've managed to come up with the following:
treeBuilder.In(
visitor.Visit(arguments[0]).AsExpression(),
treeBuilder.MethodCall("SELECT [Id] FROM [dbo].[GetCategories]", new[] {
visitor.Visit(arguments[1]).AsExpression()
}).AsExpression()
)
It's abit of a hack but it works so I guess it will do for now.