fix bug noticed in today's class