Changed to $post->ID in next version