Troubleshoot diff_drv.py