Fix type unmatching