Fixed stupid bug from testing