fix: screen flash was broken in previous commit