Fix successful run through for the tests