tried fixing the bug, still exists. Changed test implementation to the new one.