Fix issue with changes not 'staying' until page refresh