Small fix so the program runs