Fix imports from LinAlg, not Base