fix BUGs