Update setup script to match rake task changes