fix for connecting model classes.