Added helper and now model works