Fix commented out test