forgot to remove old model logic