Fixed a missing import
Merge pull request #1471 from gem/fixed-import

Fixed a missing import