Small bug fix in the last commit.