Fix preload unit tests