Fixed incorrect variable names that broke the test