Don't be dumb, use a main. Fixes execution. Update tests as well. Happy for now, but I don't like the tests.