Correction

Add missing variable and correct the file name in import statement