Drag and Drop MonoBehaviour Type in SO to Instantiate Later

In my project I am using ScriptableObjects to store information about units/buildings.

I want to use a reference to a type on the SO called MovementBehaviour, which will be instantiated on the unit GameObject when it is created to control its movement, so I can just drag one in the inspector.

I can’t figure out a way to do this. What other methods could I use for this?

Mayby you’re looking for a solution like this?

Here’s the direct link to the script on github.