fixed a bug, lucky catch