Quick fix. Stupid logic ...