remove hardcoding from test