Sorry. I just realised I am an idiot. Your first example should work.