do this on update too