Fix test coverage #12