Attaching patch with very hacky fix. Undoubtedly, you can come up with something cleaner and more general!
Attaching patch with very hacky fix. Undoubtedly, you can come up with something cleaner and more general!