Fix of logic / setup error in test