Fix marshal argument error

/cc @Proby