Added a test to make sure everything still works