FIX bug in test