fixed model to handle same size probs