Make sure State pattern tests pass after directory renaming

Just a small fix to make sure tests pass