Reverted silly change to density matrix that introduced bug