Small bug fix for the previous commit.