Bug fix for propagating ground state AND calculating excited states