Fixed typo (missing "import" on the last line)