Remove if then statements and use lava-test-case