Working on unit test.