Corrected import statements in examples

Only managed to notice `run.sh` after filing #1244, corrected missing/incorrect statements.. 
Apologies for the double PR.