forgot a bit of cleanup