Fix treecap integration

Fix dat treecap yo