Fixed another issue (why can't I code without errors!)