Unit test is now working.