Fixup custom diminish code