Remove hard coded data from test case