Fix double yield error