Fix error from parameter refactor