Fix error in remap class