Update notes for submitted changes

* Use mock.assert_called_once_with()