Added a small unit test