Fix previous patch

benchmark : 4899110