Mini patch & clean-up