Add conditional for generic files