Little fix for last commit