Use proper gradient template parameters