Fix LDM/STM edge cases