Fixing issue with User model reference.