Fix test problem for old style classes