tweaked test code