Update API timeout code