fix class in generated params type