Simplify load method of extension