Duh bug fix