Add test for functionality (fails on master)