Fix missing coverage part