fixed a bug, sorry!