Fix test for AD module

[JulienPeloton]

IMPORTANT: Please create an issue first before opening a Pull Request.
Linked to issue(s): Closes #373

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (e47443d) 54.69% compared to head (7f1f02c) 54.69%.
Report is 2 commits behind head on master.

❗ Current head 7f1f02c differs from pull request most recent head a3018b4. Consider uploading reports for the commit a3018b4 to get more accurate results

Additional details and impacted files

@@           Coverage Diff           @@
##           master     #374   +/-   ##
=======================================
  Coverage   54.69%   54.69%           
=======================================
  Files          36       36           
  Lines        1620     1620           
=======================================
  Hits          886      886           
  Misses        734      734           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.