Fix for issue #10: Fixed wrong unittest