Minor fixes for e2e