Fix type assignment