updated model to fix split