bug fixes, e2e tests