Correcting test bug