Fix for flake

Change-Id: Ib617d8202c5744b8da905fdb73e671e512c7edf4