Fix onclick listeners