slightly fixed a bug