Update tests to work with new spec gem