Realised one of the updates partially broke this -- it should work properly again now.