Super hacky bug fix!