Короче, пока некий висняк с решением. Вроде как работает твоя правка, но только в определенных случаях. Есть случаи, когда все начисто перестает работать. Попробую покопать дальше.