Fixed compatibility with other mods. ( TODO: check breaks)