fix for suggestion in PR