Fix type miss