Fix broke on-click