Fixes type-mismatch