Probably a very naive solution, but it works!