Added a change that probably should have been in the tests to begin with