Fixed problem with running tests