bug fix (very much)