Add a hack to allow using custom models (if they exist), and prevent division by 0