Add missing balance classes back