Fixing immediate bug. I hate coding on sundays