Fix flaky tests in AssignmentTest#455

nikunjagarwal321 wants to merge 1 commit intointuit:developintuit/wasabi:developfrom

nikunjagarwal321:fix-flaky-test-assignmentobjectnikunjagarwal321/wasabi:fix-flaky-test-assignmentobjectCopy head branch name to clipboard