[Fix] Bug fixes