fix bug in gmres (with return_basis=True)