updated unitTest