Update review
- Added test for the fix