IIRC, the grid fins are not strong enough to support the rocket, and reinforcing them would add too much weight to the vehicle.
The plan is to catch the second stage the same way, and the starship in flight now is the first to have mockup pins to test the aerodynamics and see if they cause issues during reentry.