Little fix on previous commit