Anyone know how to get the stock wallpaper selection back after Flashing back to OOS?

[-BoneZ-]

After using Havoc ROM and now flashing back to stock OOS, I'm not sure how to get the stock wallpaper selection back. The one I'm talking about is where you can select wallpapers by category, like Space or Earth images or Art, etc.

 

[Namolas]

https://play.google.com/store/apps/details?id=com.google.android.apps.wallpaper