Fix missing extension bug