Add test for non from gradient fix