Fixing add message unit test