Fixed warning: unknown class