Fixes preload issues in tests