Fix  bug added in last commit.