Fix unit deps