Pientä parannusta yksikkötestaukseen.