Fixing validation to use the correct file to make predictions upon