Isn't this supposed to keep the same api as the other generators? /cc @spastorino