#391 Add test