I went back and had another look at my implementation, and got it to work. Sorry I was mistaken!