Add missing assemblies to try and fix build issue