fix #2 regression