fix extension bug