Bug fix. Momentum excitations now working for non-static bath states?