minor changes to test notebooks