Fix TD parsing bug

Fixed bug in excited state parsing