Fix last commit (code for tests)