Small fix for the latest commit.