Fix tests

just a little bit more attention... 
![facepalm](http://stream1.gifsoup.com/view3/1290449/picard-facepalm-o.gif)