Fix duplicate class