Forgot to remove some tests