Fix scheduler imports