small changes to tests

Small updates to test (I see this more consistent to local / travis tests.) 
unneeded import