bug correction

better fix