Fix subsample_mosaic api