fixed a tiny bug in code