Fix lightning animation