Fix docker flaky tests