forgot a small thing (basically same commit as before, but fixed)