Help on predicting

I have watched your class in datacamp and know no one hear to address my query.

I have drop some data column, trained and fit and predict (all good). However, when I predict on the hold out data there is an error (shape is not the same).

When we drop some of our column on our model data, we need to drop some column on the hold out data?

Thanks a lot.