fix bug in test setup