Fix: Adding a test