Add missing import.
Merge pull request #329 from mkonicek/master

Add missing import.