Fix old class syntax in test