fix edge case on load