I'm assuming this is correct, it fixes the tests so they pass again
1 file changed