Is there a way to do this that doesn't require the test dataset?