Fixed the behavior of the classes and test task closures in the extension