update register to work in e2e tests