Forgot to update the tests post reorg