Fix bug on extension