Fix tests running