EZP-21245: Fixed unit tests