Fix init bug