Fix missing argument on class assignment