Add critical fixes