Small bug fix from previous commit.