Fix single expand button click function