fix changes to generative models so the right example is in the right box