small fix for last commit