Hotfix/issue 295 diagnose not working

Fixes the issue of diagnose (v1.3.0 called it --test_grad).