Fixed unit tests compilation

Fixed unit tests compilation