I think it is the removal of this wait that is causing the import tests to fail.