Small flickering fix