Complete unit test
Complete unit test'