Update per @vanzin's review