Removing small code glitch