Fix init state on load