squash with previous commit

missed some stuff