correcting expected input to match dataset