Added explicit check for ICompilationUnit. Fixes #31
Merge pull request #32 from gamerson/master

Added explicit check for ICompilationUnit. Fixes #31