Fix latest bugs