Fixed build error in unit tests