Bug  fixes