+ fix small regression