Made changes to unit test for Knight