The AI Mode homepage got a refresh too: the history icon moved to the left and there's now a handy down arrow for smoother navigation instead of the back button.
If you like trying out experimental features, Search Labs has shifted under Settings in your account menu for quicker access.
These updates are rolling out now through the Google app beta version 17.9 for Android.