(PE-6014) Fixed broken unit test