Fixed minor extension bug