Added API check to unit tests