Make unit tests work.