Fix onResize bug