tweak to previous bug fix