Solved in a different way for my use case.