Fix the setup script to work with the new package layout.