Revert "Remove tests from file"

This reverts commit 83f3a4c6122afe5ae846cde2de56ae7c59d8f54c.
Attempt to fix test