fix duplicate class issue