small changes to tests

Small updates to test (I see this more consistent to local / travis tests.) 
@syphar  I did this now.