Fix test.
Add setup/teardown methods.