Fixed small issue with some tests