Fix code vulnerability