Fixed the test that was failing. Might do a pull request.