Fix import error. Fixes LLK/scratchr2#1460
Merge pull request #1302 from LLK/issue/scratchr2/1460

Issue/scratchr2/1460