Fix undefined class