Practicing including riffs into improv?

Take a riff, lick, pattern, or whatever that you're learning and try playing only that first. Start by playing the riff instead of improvising. When the chord changes adjust the riff accordingly to fit the changes. Then slowly start changing a note here and there gradually moving on to improvisation.