Fix reload problem

Fix for #12