Fix test by addind missing columns ##