Fix
Add setup method to tests