MouldAssembly getEntity() method updated