Fixed test

Now should work both under bs and bs4