Type safe keep rules and DEX optimization