fixed bug maybe?