Handle the different models correctly