it's really pretty simple, just memorize the riff. A riff is just a motive, if one can't memorize that, he should work on ear training.