Item likes for custom post types

What’s the best way to add your item likes to custom post types? I can easily do it by adding the custom post type to the $post_types array in the show_likes function in item-likes.php, but the obvious problem is that it will be overwritten during a Kleo upgrade.

Specially I’m working on adding it to the posts footer. I tried changing do_action(‘kleo_post_footer’) to do_action(‘story_post_footer’), then adding the following to my functions.php file. (btw – fiction is the custom post type below)

add_action('story_post_footer', array(&$this, 'show_story_likes'));
function show_story_likes()
{
	$ids = str_replace(' ', '', trim(strip_tags(sq_option('likes_exclude'))));
	$ids = explode(',', $ids);
	if(in_array(get_the_ID(), $ids)) return;
	
	$post_types = array( 'post', 'attachment','fiction' );
	$post_types = apply_filters( 'kleo_likes_post_types', $post_types );
	$current_post_type = get_post_type();
	
	if( in_array( $current_post_type, $post_types ) ) {
		echo $this->do_likes();
	}
}