Fixed broken unit tests