[FIX] warning: [rawtypes] found raw type: Class