Fix BUgs