bug fix - I hope