Fix setup to match readme.