Fix params class check