Add little unit test