Fix security UTs