fix test to select columns that actually exist