fix bug (hopefully)