Fixes for first e2e test