#396 Add test