Container fix
Fixed a class name
Removed an unused import