Fix/tests

[worais]

All Submissions:

• Does your code follow the Extendables standards?
• Have you written new tests for your changes, as applicable?
• Have you successfully run tests with your changes locally?

Changes proposed in this Pull Request:

Closes # .

How to test the changes in this Pull Request:

Other information:

• Have you checked to ensure there aren't other open Pull Requests for the same update/change?

Changelog entry

Enter a summary of all changes on this Pull Request. This will appear in the changelog if accepted.