Bug fixesg