Add functionality to replicate #336