fix old class expectation in specs