fixing tests to match new specs