Small modification to fips generator