added missing import statement
Merge pull request #165 from daniellemayne/develop

added missing import statement