Fix mismatch error