fixed issue with first test not working after going through assignment