Fix reload bug