bug fix... sorta