Fixing actual bug