Avoid unused import for enum
Merge pull request #166 from Colin-b/patch-1

Avoid unused import for enum