Minor bug fix from the previous push