bug fixes... oops