Fix unit test class name
Add unit test