Revert "Remove test"

This reverts commit ef7ad5d34ae7a889d4eb0972f6330b387434b184.
Fix test