Minor fix to remove unnecessary container