small fix for last commit 2