Fix minor bug commited in last commit.