couple of modifications to the test: