fix for import statement, unit tests failing in codeship