Replace magnetar setup with a source term which gets the energy right