[UPDATE] add little more code to enable code coverage