Trying to fix the mismatch