#381 Add tests