add missing fix in previous commit