Prepare for custom User model