Properly fix the last bug, forgot a case