Add missing mass driver stuff