Clean up loadmodel decorator

I wasn't aware of class decorator semantics when I wrote it.