Added explicit check for ICompilationUnit. Fixes #31