Oops, forgot to remove an import I used while testing
Change README