Small change to init