removing the unnecessary tests to avoid confusion during demo