Issue 8 review2

Please fix this import line for Note to correct one.