#367 Add test