fixing bug whoops sorry