Altered stories to match new test class