corrected wrong variable name; added missing import