Work on test.
Fix test.