Missing restore() in tests