Added missing use statement.
Fixed bug related to new LazyExtension class.