Little fix to previous commit