fixed a bug, it should work now