Remove test method and add todo for snapshot implementation