small fix for latest regression