fix small problem in previous commit