Fixing test framework after name change