Attempting to fix type error again

Works fine locally...