Small changes I forgot in the last commit.