Fixed another import
Merge pull request #1443 from gem/source-and-logictree

Fixed another import