fix f**king bug