forgot a small fix