fix staging breakage