Added missed fixed for previous commit