Fixing unit test equality check