fixed a slight bug