Fixing the tests so they can run with prove