hw6 bugfix