Removed import from TestContainer that caused compilation error