Forgot to update setup script