Updated for compatibility with YiiBooster