Minor changes to Julia test framework and tests

- Getting rid of unused var
- Refactored test slightly