better fix

this fix should not introduce problems