Ahhh bug fix!