Fixing start/end party button states