Fix unitTest