Fix test message.
Add test to assert error message. Close #2