minor changes to test functions