use proper unit tests