fixed broken unitests