fix a small problem