Improve behaviour with loading/unloading

Signed-off-by: Marcel Verpaalen <marcel@verpaalen.com>