fix a related bug