Fixed test based on eugene's feedback