Fixing use statement
Merge pull request #440 from ciaranmcnulty/master

Fixing use statement