Fixed type errors in unit tests.