[BUG] Fix version compat problem