Small fix to the previous commit.