Fixing type mismatch