Forgot to adjust some files