The test is still broken, there are print statements... this is an ongoing WIP and will get squashed before submitted the PR.