Added in files to assert against and everything works as expected =D