fix tests bug
added comments to fix