JS bug fix..sorta