small fix.. had to safe returns from the main function somehow