fixing unit tests for notation shift