Little fix to the previous commit