*Actually* fix the build

Stray import lines for the loss.