Fixes adding sample tests