Temporary fix for crash between EDX and ExtraTrees