fix test set-up