Minor: fix niggle with import statement

Changed import to longer form to make later additions easier