fix output expectation