Use should gem to clean up more tests.