I think we should loko into anther option that is not hardcoding the types in the parent class