Fix flaky initialize handler tests

Fix flaky initialize handler tests #1095