Fixed unit testing build error