Fix pytest break on exception every time (#740)

Fix pytest break on exception every time (#740) #398