small fix and test again