Adds missing code removed on last commit.