Add the script code to the extension code

[rafaelgomesxyz]

Using a conditional to find out if the user is executing the script or the extension and then running the correspondent code will prevent future pasting mistakes.