#123 Fix for pull request. Added import.