Fix type annotation #9