Fix running bug