Fix misc. bugs