TST+BUG Fix bug in test