Improving #10 - Fixing wrong test