fix bug -.-