Add some missing capability checks