fix return model