Add tests, fix condition parameter type