#368 Add test