Hi Alex,
There is no 100% guarantee for proper indexing of the relationship fields.
The code seems to be all right to me, so can you please make sure that you are using the index table engine, and if it is indeed enabled. Also, after adding this custom code make sure to re-create the index table, so that the relationship fields are picked up.