tiket:10 fix ValueError(remove test output from model)