Fix one-click bug