Labour is not the only thing capitalists need from the workers, they're also at the same time commodity consumers. Even when reaching an advanced level of automation, the capitalists will need people to purchase their products which is where UBI enters the picture. Interestingly, the "UBI" class could actually strike by not consuming stuff.

Anyways i think you're just overthinking, the UBI in the imperial core is only a labour subsidy.