Fix tool shed unit tests I broke with my last commit.