Fix init function