fixing failed test after intial submission of assignment