Forgot to revert some changes