Fixed another small bug in the init