Revert "Remove test"

This reverts commit aceac8615fe0338094969dd3d897ae26b0e5d962.
Try to fix test