fixing wrong model sizes