Added a test unit test