My Nieco unit will light but then shuts off.

[ectofix]

Along with a good wire connection to the flame sensor, make sure the flame sensor has a good, clean ground connection to the burner.  That ground is often a tab protruding from the from the flame sensor mount.  It must contact the burner when you reinstall it.