updated test set up