Fix: minor bug into the test.
Update: added comments to make exercise scope more clear.