Fixing the test to test actual code thats run