Issue #9. Updated the test to make it more type-checking.