Forgot to remove some unused code