Updating tests to actually pass

Should probably be testing per level, but will fix in the morning