Fix assert_called_once
Fix some comments :))