fix bug, hopefully