Just come across another post here that has a fix for anyone searching for this
Switching the lazy load to eager resolved this for me