fixed #1

Add unit test