Fix container mock assertion