Fixed indent (tab vs. spaces) and added missing import.