Manage to get e2e test to work

SUCH A WIN!