Add missing import
TreeGA generator is now const