Fix reverting something I had for testing