Fixed the overlay handling for --model-small and --model-large