Get the missing unit tests to work.