The CPU code works in the experiment but not in the test. Trying to fix
it.