In rough_surface_generator: updated some args