small fix for previous commit