fixed build break after class renaming