While Goku had eventually become Super Saiyan God, Vegeta was initially still stuck with Super Saiyan 2. However, he then started training with Whis, and eventually unlocked Super Saiyan Blue.