[PLAT-4541] Modifying test to reflect changes