minor: Fix type errors