Fixing onEnable()