Fixing unittest after adding PlainContainer