Added missin unit test