Increase code coverage by adding some missing test