Fix the mass driver again...

Closes #12