Fixed lost changes from the last commit.