Fix reload issue