Fixed bug created by vulnerability fix