XD-244 Code Review / Add test