You are overriding variable $args. Please change your arguments name to something else. Also you do not need to provide child messages ids.
count is cached for performance. Please use same function in hook fep_message_count_query_args also. It will be hard to maintain count properly if you do not invalid cache when ref ids change for any user.