fixed a bug maybe