Small fix to previous commit.