fix missing edit from previous commit