Fix small glitch