Returning to Canada Thomson worked for a short time at the National Film Board of Canada before being invited to served as Associate Private Secretary to Canadian Prime Minister Louis St. Laurent where he remained until 1958 .