fix bug, silly me