Fix problems preventing plugin from loading
Small tweak