fixed unit testing