Toolkit fixes