Forgot the actual unit test changes