Add test for #336