Fix treecap integration
Merge pull request #152 from cexikitin/master

Fix treecap integration