You should probably move this to the actual model.