Fix running tests from console