Fix onclick function