Fix issue #681

[rubencalje]

Short Description

Fixes issue #681

Checklist before PR can be merged:

• closes issue #xxxx
• is documented
• Format code with Black formatting
• type hints for functions and methods
• tests added / passed
• Example Notebook (for new features)
• Remove output for all notebooks with changes

[codacy-production]

Coverage summary from Codacy

See diff coverage on Codacy

Coverage variation	Diff coverage
✅ +0.11% (target: +0.00%)	✅ 100.00%

Coverage variation details

	Coverable lines	Covered lines	Coverage
Common ancestor commit (9c76689)	5833	4358	74.71%
Head commit (b3a179d)	5858 (+25)	4383 (+25)	74.82% (+0.11%)

Coverage variation is the difference between the coverage for the head and common ancestor commits of the pull request branch: <coverage of head commit> - <coverage of common ancestor commit>

Diff coverage details

	Coverable lines	Covered lines	Diff coverage
Pull request (#682)	1	1	100.00%

Diff coverage is the percentage of lines that are covered by tests out of the coverable lines that the pull request added or modified: <covered lines added or modified>/<coverable lines added or modified> * 100%

See your quality gate settings    Change summary preferences

_{You may notice some variations in coverage metrics with the latest Coverage engine update. For more details, visit the documentation}