Another small fix related to last commit.