fixed broken unittest