minor changes to test