tweaks & remove test