minor fix in the extension