Type mismatch fixed