Finish writing unit tests.