made the test pass for demo