BUg fixes