Works as expected now